#pragma once

#include <iostream>
#include <string>
#include <queue>
#include <pthread.h>
#include "Task.hpp"
const static int defaultcap = 5;
template <class T>
class BlockQueue
{
private:
    bool IsFull()
    {

        return _block_queue.size() == _max_cap;
    }
    bool IsEmpty()
    {
        return _block_queue.empty();
    }

public:
    BlockQueue(int cap = defaultcap)
        : _max_cap(cap)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_p_cond, nullptr);
        pthread_cond_init(&_c_cond, nullptr);
    }
    // 将数据带出去的接口：
    void Pop(T *out)
    {
        pthread_mutex_lock(&_mutex);
        if (IsEmpty())
        {
            pthread_cond_wait(&_c_cond, &_mutex); // 等待唤醒
        }
        // 1.没有空 || 2.被唤醒了
        // 说明一定有数据，因此要消费
        *out = _block_queue.front();
        _block_queue.pop(); // 消费者消费一个数据，就一定知道队列不是满的
        // 生产者最清楚队列一定不为空

        pthread_mutex_unlock(&_mutex);
        pthread_cond_signal(&_p_cond); // 生产者没睡着，此操作无效
    }
    // 入队列
    void Equeue(const T &in)
    {
        pthread_mutex_lock(&_mutex);
        if (IsFull()) // 队列是否满了
        {
            // 生产这不能生产，必须等待
            // 在临界区里面在等待，但锁没有释放，即便等待了，但是锁没有释放，注定消费者pop的时候也无法拿到锁，死锁
            pthread_cond_wait(&_p_cond, &_mutex); // 函数返回的时候，不还在临界区吗，并且锁不是已经被释放了吗
            // 等待唤醒
            // pthread_cond_wait在被调用的时候，除了让自己继续排队等待，还会自己释放传入的锁
        }
        // 1.没有满 || 2.被唤醒了//因此就可以生产到阻塞队列（还没解锁，数据不会被拿走，一定是有数据的）
        _block_queue.push(in);
        pthread_mutex_unlock(&_mutex);
        // 有数据了，就要让消费者消费
        pthread_cond_signal(&_c_cond); // 消费者没睡着，此操作无效，

        // 如果，生产者只有一个，消费者有两个，当两个消费者都在等待的时候，生产者采用
        // pthread_cond_broadcast()唤醒线程怎么办（此时只有一个能提供给消费者消费的资源
    }

    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_p_cond);
        pthread_cond_destroy(&_c_cond);
    }

private:
    std::queue<T> _block_queue; // 临界资源
    int _max_cap;               // 队列最大值
    pthread_mutex_t _mutex;     // 定义一把锁
    pthread_cond_t _p_cond;     // 给生产者提供的条件变量
    pthread_cond_t _c_cond;     // 给消费者提供的条件变量
};
